from fastapi import APIRouter

# from .knowledge_chat import knowledge_chat
from .knowledge_chat_v2 import knowledge_chat
from .keyword_chat import keyword_chat
from .emergency_chat import emergency_chat
from .single_file_chat import single_file_chat
from .pre_meeting_chat import pre_meeting_chat
from .accident_chat import accident_chat
from .risks_identification_chat import risks_identification_chat

router = APIRouter()

router.post("/knowledge/keyword_chat", tags=['knowledge chat'], summary="关键词检索")(keyword_chat)
router.post("/knowledge/chat", tags=['knowledge chat'], summary="ai检索")(knowledge_chat)
router.post("/emergency/chat", tags=['knowledge chat'], summary="应急演练")(emergency_chat)
router.post("/single_file/chat", tags=['knowledge chat'], summary="单文件chat")(single_file_chat)
router.post("/pre_meeting/chat", tags=['knowledge chat'], summary="班前会/组前会")(pre_meeting_chat)

router.post("/accident/chat", tags=['accident chat'], summary="安全事故案例分析")(accident_chat)
router.post("/risks_identification/chat", tags=['risks identification'], summary="图片风险识别")(risks_identification_chat)